from rest_framework import serializers
from .models import Predict, Train


class TrainModelSerializer(serializers.ModelSerializer):
    """
    训练信息序列化器
    """
    class Meta:
        model = Train
        fields = '__all__'


class PredictModelSerializer(serializers.ModelSerializer):
    """
    预测信息序列化器
    """
    class Meta:
        model = Predict
        fields = '__all__'
